# coding: utf-8

import os
import matplotlib.pyplot as plt
import numpy as np
import matplotlib

font = {"family": "Arial", "weight": "normal", "size": 14}
matplotlib.rc("font", **font)
LineWidth = 3


# ECCV 2024: WeConvene: Learned Image Compression with  Wavelet-Domain Convolution and Entropy Model
bpp = [0.130, 0.156, 0.224, 0.311, 0.435, 0.622]
msssim = [14.6440, 15.3225, 16.5814, 17.8566, 19.0953, 20.4474]
weconvene, = plt.plot(bpp, msssim, "k-o", linewidth=LineWidth, label='WeConvene')


# NOTE: 绘图
plt.legend(handles=[weconvene], loc=4)
plt.grid()
plt.xlabel('Bpp')
plt.ylabel('MS-SSIM')
plt.title('Tecnick dataset')
plt.savefig('tecnick_msssim.png', format='png', dpi=600, bbox_inches='tight')
